how to create colour game in Python